import warnings
warnings.filterwarnings("ignore", category=Warning)

from utils import EmbeddingLoader

# 用户输入的参数
redis_host = "localhost"
redis_port = 6379
# graph_name = "CoraGraph_LINE"
# graph_name = "CoraGraph"
# graph_name = "CoraGraph_Lap"
# graph_name = "CiteseerGraph"
# graph_name = "CiteseerGraph_LINE"
# graph_name = "CiteseerGraph_Lap"
# graph_name = "PubmedGraph"
# graph_name = "PubmedGraph_Lap"
# graph_name = "DblpGraph"
# graph_name = "DblpGraph_Lap"
# emb_file_path = "../data/Cora.emb"
# emb_file_path = "../data/Cora_spectral.emb"
# emb_file_path = "../data/Cora_Lap.emb"
# emb_file_path = "../data/Citeseer_spectral.emb"
# emb_file_path = "../data/Citeseer.emb"
# emb_file_path = "../data/Citeseer_Lap.emb"
emb_file_path = "../data/Pubmed_spectral.emb"
# emb_file_path = "../data/Pubmed_Lap.emb"
# emb_file_path = "../data/Dblp_spectral.emb"
# emb_file_path = "../data/Dblp_Lap.emb"
batch_size = 1000
graph_name = "Pubmed"
# 加载和更新嵌入到 RedisGraph
embedding_loader = EmbeddingLoader(
                    redis_host, 
                    redis_port, 
                    graph_name, 
                    emb_file_path, 
                    batch_size
                )
embeddings = embedding_loader.load_embeddings()
embedding_loader.update_graph_embeddings(embeddings)
embedding_loader.verify_embeddings(test_id=633)